package com.example.tests;

import static com.example.tests.ContactsDataGenerator.generateRandomContacts;
import static com.example.tests.GroupDataGenerator.generateRandomGroups;

import java.io.FileReader;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;

import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.DataProvider;

import com.example.fw.ApplicationManager;

public class TestBase
{
	protected static ApplicationManager app;
	private int checkCounter;
	private int checkFrequency;

	@BeforeTest
	public void setUp() throws Exception
	{
		String configFile = System.getProperty("configFile", "application.properties");
		Properties properties = new Properties();
		properties.load(new FileReader(configFile));
		app = new ApplicationManager(properties);
		checkCounter = 0;
		checkFrequency = Integer.parseInt(properties.getProperty("check.frequency"));
	}

	protected boolean timeToCheck()
	{
		checkCounter++;
		System.out.println(checkCounter);
		if (checkCounter > checkFrequency)
		{
			checkCounter = 0;
			return true;
		}
		else
		{
			return false;
		}
	}

	@AfterTest
	public void tearDown() throws Exception
	{
		app.stop();
	}

	@DataProvider
	public Iterator<Object[]> randomValidGroupGenerator()
	{
		return wrapGroupDataForDataProvider(generateRandomGroups(5)).iterator();
	}

	@DataProvider
	public Iterator<Object[]> randomValidContactGenerator()
	{
		return wrapContactsDataForDataProvider(generateRandomContacts(5)).iterator();
	}

	public static List<Object[]> wrapGroupDataForDataProvider(List<GroupData> groups)
	{
		List<Object[]> list = new ArrayList<Object[]>();
		for (GroupData group : groups)
		{
			list.add(new Object[] { group });
		}
		return list;
	}

	protected List<Object[]> wrapContactsDataForDataProvider(List<ContactData> contacts)
	{
		List<Object[]> list = new ArrayList<Object[]>();
		for (ContactData contact : contacts)
		{
			list.add(new Object[] { contact });
		}
		return list;
	}

}
